package com.improvess.rna.som.knn;

import java.io.File;
import java.io.IOException;
import java.util.Map;

import Jama.Matrix;

import com.improvess.rna.som.DataSet;
import com.improvess.rna.som.loaddata.LoadData;

public class TesteKNN {
	
	public static void main(String[] args) {
		
		try {
			DataSet conjTreinamento = LoadData.loadConjuntoDeTreinamento(new File("train1.txt"));
			
			DataSet conjTeste = LoadData.loadConjuntoDeTreinamento(new File("train2.txt"));
			
			double[][] treinamento = conjTreinamento.getRegistros();
			double[][] teste = conjTeste.getRegistros();
			
			Matrix[] matrixTreinamento = new Matrix[treinamento.length];
			for (int i = 0; i < treinamento.length; i++) {
				double[] x = treinamento[i];
				matrixTreinamento[i] = new Matrix(new double[][] { x });
				matrixTreinamento[i] = matrixTreinamento[i].transpose();
			}
			
			Matrix[] matrixTeste = new Matrix[teste.length];
			for (int i = 0; i < teste.length; i++) {
				double[] x = treinamento[i];
				matrixTreinamento[i] = new Matrix(new double[][] { x });
				matrixTreinamento[i] = matrixTreinamento[i].transpose();
			}
			
			Knn knn = new Knn(5);
			knn.setTrainingSet(null);
			
			for(int i = 0; i< matrixTeste.length;i++){
				Map<String, Double> probPorClasse = knn.processar(matrixTeste[i]);
				System.out.println("Classe do elemento: ");
			}
		} catch (IOException e) {
			e.printStackTrace();
		}
		
	}

}
